# -*- encoding: utf-8 -*-
"""

PyCharm 复指数序列信号的产生
2024年10月22日20时27分
by wenyang

文件主要内容:


"""
import numpy as np
import matplotlib.pyplot as plt

def main():
    # 生成采样的信号
    # 设置采样的频率
    Fs = 1
    n = np.arange(-10, 10, 1 / Fs)
    # 生成信号
    # 设置幅值
    A = 3
    # 设置指数
    a = 0.5 + 0.5j
    # 定义一个阶跃信号
    y1 = np.array([1 if i >= 0 else 0 for i in n])
    # print(y1)
    # plt.plot(n, y1)
    y = A * y1 * a ** n
    # 绘制图像
    plt.stem(n, y)
    plt.title("signal")
    plt.xlabel("n")
    plt.ylabel("y")
    plt.show()


if __name__ == "__main__":
    main()
